package com.example.mq_sm.mqserver.datacenter;

import com.example.mq_sm.MqSmApplication;
import com.example.mq_sm.mqserver.core.Binding;
import com.example.mq_sm.mqserver.core.Exchange;
import com.example.mq_sm.mqserver.core.ExchangeType;
import com.example.mq_sm.mqserver.core.MSGQueue;
import com.example.mq_sm.mqserver.mapper.MetaMapper;

import java.io.File;
import java.util.List;

/**
 * 通过这个类，来整合数据库操作
 */
public class DataBaseManager {

    // 这里的 metaMapper 还没有注入 bean 所以在后续的使用中一定会报一个空指针异常
    // 一般来说我们是在这进行一个 @Autowird 的注释进行一个注入
    // 但是要是这样 DataBaseManager 中也需要进行注释，将其也设置为 bean
    // 不过后续需要进行一个手动管理，这样就行不通了，所以需要手动构造
    private MetaMapper metaMapper;

    // 针对数据库进行初始化
    public void init(){
        // 手动获取到 metaMapper
        metaMapper = MqSmApplication.context.getBean(MetaMapper.class);

        // 首先先判断是否存在库表，如果不存在就建库建表
        if (!checkDBExists()){
            // 需要先创建一个 data 目录
            File dateDir = new File("./data");
            if (!dateDir.exists()){
                dateDir.mkdirs();
            }
            // 进行一个建表操作
            createTable();
            // 建完表后设置一个默认的值
            createDefaultData();
        }
    }

    // 判断表是否存在，就看 meta.db 文件是否存在，这个文件的路径和文件名都是在配置文件中配置好的
    private boolean checkDBExists() {
        File file = new File("./data/meta.db");
        return file.exists();
    }

    // 库在项目启动的时候自动就会建好，所以在这里只需要创建表即可
    private void createTable() {
        metaMapper.createExchangeTable();
        metaMapper.createQueueTable();
        metaMapper.createBindingTable();
        System.out.println("[DataBaseManager] 创建表完成");
    }

    // 给数据库表中添加一个默认的数据
    // 这里的设置一个基本数据是设置一个匿名的交换机
    // RabbitMQ 是这样进行设置的
    private void createDefaultData() {
        Exchange exchange = new Exchange();
        exchange.setName("");
        exchange.setType(ExchangeType.DIRECT);
        exchange.setDurable(true);
        exchange.setAutoDelete(false);
        metaMapper.insertExchange(exchange);
        System.out.println("[DataBaseManager] 默认数据设置完成");
    }

    // 封装其他的数据库操作
    public void insertExchange(Exchange exchange){
        metaMapper.insertExchange(exchange);
    }

    public List<Exchange> selectAllExchanges(){
        return metaMapper.selectAllExchanges();
    }

    public void deleteExchange(String exchangeName){
        metaMapper.deleteExchange(exchangeName);
    }

    public void insertQueue(MSGQueue queue){
        metaMapper.insertQueue(queue);
    }

    public List<MSGQueue> selectAllQueues(){
        return metaMapper.selectAllQueues();
    }

    public void deleteQueue(String queueName){
        metaMapper.deleteQueue(queueName);
    }

    public void insertBinding(Binding binding){
        metaMapper.insertBinding(binding);
    }

    public List<Binding> selectAllBindings(){
        return metaMapper.selectAllBindings();
    }

    public void deleteBinding(Binding binding){
        metaMapper.deleteBinding(binding);
    }

    public void deleteDB() {
        File file = new File("./data/meta.db");
        boolean ret = file.delete();
        if (ret) {
            System.out.println("[DataBaseManager] 删除数据库文件成功!");
        } else {
            System.out.println("[DataBaseManager] 删除数据库文件失败!");
        }

        File dataDir = new File("./data");
        // 使用 delete 删除目录的时候, 需要保证目录是空的.
        ret = dataDir.delete();
        if (ret) {
            System.out.println("[DataBaseManager] 删除数据库目录成功!");
        } else {
            System.out.println("[DataBaseManager] 删除数据库目录失败!");
        }
    }
}
